# -*- coding:utf-8 -*-
"""
使用wind提取股票的相关信息

从wind中提取的字段有：
'SEC_NAME', 股票简称
'PRENAME', 证券曾用名
'SEC_NAME1', 证券简称（支持历史）
'COMP_NAME', 公司中文名称
'CHAIRMAN', 法定代表人
'BOARDCHAIRMEN', 董事长
'CEO', 总经理
'holder_name',大股东名称
'holder_controller', 实际控制人名称
'holder_liqname', 流通股东名称

通过这些字段构建股票关系图谱
"""
from WindPy import *
import os
import pickle

w.start()

# 读取股票列表信息
stockList = []
with open('../data/stockList.csv', 'r', encoding='utf-8') as f:
    for each in f.readlines():
        stockList.append(each.strip())

stockStr = ','.join(stockList)
path = '../data/windData/'
# 读取每年 5.1/9.1/11.1的数据
for year in range(2012, 2018):
    for date in ['0501', '0901', '1101']:
        year_date = str(year) + date
        print(year_date)
        data = w.wss(stockStr, "sec_name,prename,sec_name1,comp_name,chairman,boardchairmen,ceo,holder_name,holder_controller,holder_liqname","tradeDate="+year_date+";order=0")
        # 将数据进行序列化
        windFile = open(os.path.join(path, year_date+'.pkl'), 'wb')
        pickle.dump(data, windFile, -1)